package com.platform.system.logic;

import com.platform.system.enums.ModelTypeEnum;
import com.platform.system.model.BaseModelHandle;
import com.platform.system.model.MiniMaxModel;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.Map;

/**
 * @author Huangkai
 */
@Slf4j
@Component
@Data
public class AiLogic {

    private Map<Integer, BaseModelHandle> map = new HashMap<>();

    @PostConstruct
    public void build(){
        map.put(ModelTypeEnum.MiniMax.code, getMiniMaxModel());
    }

    public BaseModelHandle getMiniMaxModel(){
        BaseModelHandle baseModelHandle = new MiniMaxModel();
        return baseModelHandle;
    }
}
